from django.contrib.auth.mixins import LoginRequiredMixin
from django.http import JsonResponse

"""
登录验证  继承LoginRequiredMixin重写handle_no_permission方法
"""


class LoginMixin(LoginRequiredMixin):
    # 重写方法  如果没有查询到用户就返回未登录
    def handle_no_permission(self):
        return JsonResponse({'code': 400, 'errmsg': '未登录'})
